package leetcode.alg.demo;

import java.util.Stack;

/**
 * 定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中，
 * 调用 min、push 及 pop 的时间复杂度都是 O(1)。
 *
 * 示例:
 * MinStack minStack = new MinStack();
 * minStack.push(-2);
 * minStack.push(0);
 * minStack.push(-3);
 * minStack.min();   --> 返回 -3.
 * minStack.pop();
 * minStack.top();      --> 返回 0.
 * minStack.min();   --> 返回 -2.
 *  
 * 提示：
 * 各函数的调用总次数不超过 20000 次
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class MinStack {
    public static void main(String[] args){
        MinStack minStack = new MinStack();
        minStack.push(-2);
        minStack.push(0);
        minStack.push(-3);
        System.out.println(minStack.getMin());
        minStack.pop();
        System.out.println(minStack.top());
        System.out.println(minStack.getMin());
    }

    Stack<Integer> stack_A;
    Stack<Integer> stack_B;

    public MinStack() {
        stack_A = new Stack<>();
        stack_B = new Stack<>();
    }

    public void push(int x) {
        stack_A.push(x);
        if(stack_B.empty() || stack_B.peek() >= x){
            stack_B.push(x);
        }
    }

    public void pop() {
        if(stack_A.pop().equals(stack_B.peek())){
            stack_B.pop();
        }
    }

    public int top() {
        return stack_A.peek();
    }

    public int getMin() {
        return stack_B.peek();
    }
}
